from llm import NovelLLM
from util import writeFile
from .base import BaseNode

class Step1Node(BaseNode):
    def run(self, state):
        prompt = """
#Task
一句话故事概括
- 用20-25字提炼核心，聚焦1-2个主角及其任务，不含结局。
- 例如：“一名高中男生与小偷为阻止灾难，冒险进入燃烧的大楼。
"""
        result = NovelLLM().call_llm(prompt, caller_name="Step1")
        state['step1_output'] = result

        writeFile("output/step1.txt", result)

        print(f"Step1 完成故事梗概生成")
        return state